write a while loop code on python